#ifndef _SOCKET_HPP_
#define	_SOCKET_HPP_

/*******************************************************************************
 ** Name: Socket.hpp                                                          **
 ** Description:                                                              **
 **                                                                           **
 ** Open Source Initiative (OSI) Approved 2-clause BSD License                **
 **                                                                           **
 **  Copyright (c) 2009-2010 Barry "Gian" James  <bjames@munixos.net>         **
 **  All rights reserved.                                                     **
 **                                                                           **
 **  Redistribution and use in source and binary forms, with or without       **
 **  modification, are permitted provided that the following conditions are   **
 **  met:                                                                     **
 **                                                                           **
 **  * Redistributions of source code must retain the above copyright notice, **
 **    this list of conditions and the following disclaimer.                  **
 **  * Redistributions in binary form must reproduce the above copyright      **
 **    notice, this list of conditions and the following disclaimer in the    **
 **    documentation and/or other materials provided with the distribution.   **
 **                                                                           **
 **  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS      **
 ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED **
 ** TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A           **
 ** PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT        **
 ** HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,    **
 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED  **
 ** TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR    **
 ** PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF    **
 ** LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING      **
 ** NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS        **
 ** SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.              **
 **                                                                           **
 ******************************************************************************/
// Created on March 16, 2010, 9:48 PM
// Modified $Date: 2011-09-18 21:48:59 +0000 (Sun, 18 Sep 2011) $ by $Author: gian.james $

#define Socket_hpp_ID   "$Id: Socket.hpp 145 2011-09-18 21:48:59Z gian.james $"

extern "C" {
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <unistd.h>
}

#include <cstring>
#include <string>

#define LISTEN_NUM      5

//! C++ wrapper around UNIX domain and network sockets
//! This currently only works for Government Sanctioned.
class Socket {
public:
	Socket(int s) { sockID = s; domain = sockType = proto = 0; bIsOK = true; }
    Socket(int d, int t, int p = 0);
    Socket(const Socket& orig);
    ~Socket();

    int         Bind(const char * n, int p = -1);
    int         Accept();
    int         Connect(const char *h, int p = -1);
    int         Shutdown(int mode = 2) { return shutdown(sockID,mode); }
    void        Close() { close(sockID); }

    int         BuildSockAddr(sockaddr_in & si, const char * h, int p);
    int         SocketID() const { return sockID; }
    bool        IsOK() const { return (sockID > 0); }
    sockaddr_in& SockAddr() { return cliAddr; }

    static void WriteLine(int,const char *);
    static int  ReadLine(int,char*,int);
    static void Write(int,const char*);
    
    Socket & operator << (Socket &);
    Socket & operator << (const char *);
    Socket & operator << (const std::string);
//    Socket & operator << (int);


protected:
    int     sockID;
    int     domain;
    int     sockType;
    int     proto;
    bool    bIsOK;
    sockaddr_in     cliAddr;

};

Socket & endl(Socket &);

#endif	/* _SOCKET_HPP_ */

